python one line if elif